from lenet import LeNetModel
from train_set import train

# 加载模型
model = LeNetModel()
# 开始训练
train(model)